import unittest, ddt

"""
ddt 使用步骤:
1. 在类上面添加装饰器:ddt.ddt
2. 在测试方法上面添加装饰器:ddt.data
"""
list1 = [1, 2, 3, 4, 5, 6, 7, 8, 9]

# 写一个测试类
@ddt.ddt
class MyTest(unittest.TestCase):
    # @ddt.data(3)                                            # 传递了一个测试数据
    # @ddt.data(3, 1, 2, 3, 4, 5, 6, 7, 8)                    # 传递了一组测试数据
    @ddt.data(*list1)                                         # 传递了提前准备好的一组测试数据
    def test_bb1(self, value):
        print(f"\n====>{value}")
        # 断言
        assert value == 2, "这个数不是2"


if __name__ == '__main__':
    unittest.main(verbosity=2)
